package org.dromara.common.tenant.handle;

import cn.hutool.core.collection.ListUtil;
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.NullValue;
import net.sf.jsqlparser.expression.StringValue;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.tenant.helper.TenantHelper;
import org.dromara.common.tenant.properties.TenantProperties;

import java.util.List;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：自定义租户处理器
 */
/**
 * PlusTenantLineHandler 类实现了 TenantLineHandler 接口，用于处理特定租户的行数据操作
 * 该类提供了定制化的行处理逻辑，旨在为特定租户提供数据处理功能
 */
@Slf4j
@AllArgsConstructor
public class PlusTenantLineHandler implements TenantLineHandler {

    /**
     * 租户 配置属性
     */
    private final TenantProperties tenantProperties;

    /**
     * 获取租户ID的表达式
     *
     * 此方法用于获取当前上下文中的租户ID如果无法获取有效的租户ID，则返回一个空值表达式
     *
     * @return Expression 租户ID的表达式，有效时返回StringValue，无效时返回NullValue
     */
    @Override
    public Expression getTenantId() {
        // 尝试获取当前的租户ID
        String tenantId = TenantHelper.getTenantId();
        // 当租户ID为空或空白时，记录错误日志并返回空值表达式
        if (StringUtils.isBlank(tenantId)) {
            log.error("无法获取有效的租户id -> Null");
            return new NullValue();
        }
        // 返回固定租户 返回获取到的租户ID作为字符串值
        return new StringValue(tenantId);
    }

    /**
     * 判断是否忽略指定的表
     *
     * @param tableName 表名
     * @return 如果表需要被忽略，则返回true；否则返回false
     */
    @Override
    public boolean ignoreTable(String tableName) {
        // 获取当前租户ID
        String tenantId = TenantHelper.getTenantId();
        // 判断是否有租户
        if (StringUtils.isNotBlank(tenantId)) {
            // 不需要过滤租户的表
            List<String> excludes = tenantProperties.getExcludes();
            // 非业务表  初始化非业务表列表，这些表不会被忽略
            List<String> tables = ListUtil.toList(
                "gen_table",
                "gen_table_column"
            );
            // 将不需要过滤的租户表添加到非业务表列表中
            tables.addAll(excludes);
            // 检查传入的表名是否在忽略列表中
            return tables.contains(tableName);
        }
        // 如果没有租户，或者传入的表名不在忽略列表中，则不忽略该表
        return true;
    }

}
